"""File that contains PES object data generated via functions in vibrant."""

# pylint: skip-file

import numpy as np

# Reference copied from vibrant HDF5 files
# mol = H2S
# coords =[[0.0, -1.0, -1.0], [0.0, 1.0, -1.0], [0.0, 0.0, 0.0]]
# localize = True

taylor_1D = np.array(
    [
        [1.01030577e-05, -4.47405417e-05, -1.10879929e-05],
        [-7.12792255e-05, -1.38984154e-03, 1.84471155e-04],
        [-7.12793370e-05, -1.38984152e-03, 1.84471151e-04],
    ]
)

taylor_2D = np.array(
    [
        [
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
        ],
        [
            [
                -5.31504581e-05,
                1.26195669e-04,
                4.59577981e-04,
                -3.19150467e-05,
                -2.12745957e-04,
                2.58007346e-05,
            ],
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
        ],
        [
            [
                -5.31505500e-05,
                1.26195707e-04,
                4.59577965e-04,
                -3.19150532e-05,
                -2.12745955e-04,
                2.58007383e-05,
            ],
            [
                -1.27269515e-04,
                1.13495645e-05,
                1.13497539e-05,
                -1.88516283e-06,
                1.92460286e-06,
                -1.88519488e-06,
            ],
            [
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
                0.00000000e00,
            ],
        ],
    ]
)

taylor_3D = np.array(
    [
        [
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
        ],
        [
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
        ],
        [
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [-1.91690218e-04, 4.04877496e-05, -1.76014298e-05, -1.76014291e-05],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
            [
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
            ],
        ],
    ]
)

freqs = [0.00589689, 0.01232428, 0.01232428]

uloc = [
    [1.0, 0.0, 0.0],
    [0.0, -0.70710712, -0.70710644],
    [0.0, -0.70710644, 0.70710712],
]

# Pre-sorted
reference_taylor_bosonic_coeffs = np.array(
    [
        -0.0014741495669876705,
        -0.0014741495669876705,
        -0.0014741495436878212,
        -0.0014741495436878212,
        -0.0013076515364979621,
        -0.0013076515364979621,
        -0.00130765145178359,
        -0.00130765145178359,
        -0.0004913831889958902,
        -0.0004913831889958902,
        -0.0004913831812292737,
        -0.0004913831812292737,
        -0.00021274595723421176,
        -0.00021274595487619645,
        (-0.00011795343850034418 + 0j),
        (-0.00011795343850034418 + 0j),
        -0.00010637297861710588,
        -0.00010637297861710588,
        -0.00010637297861710588,
        -0.00010637297861710588,
        -0.00010637297743809822,
        -0.00010637297743809822,
        -0.00010637297743809822,
        -0.00010637297743809822,
        -6.646252603025534e-05,
        -6.646252603025534e-05,
        -6.646252603025534e-05,
        -6.646252603025534e-05,
        -5.318648930855294e-05,
        -5.318648930855294e-05,
        -5.318648930855294e-05,
        -5.318648930855294e-05,
        -5.318648871904911e-05,
        -5.318648871904911e-05,
        -5.318648871904911e-05,
        -5.318648871904911e-05,
        -4.745451061101081e-05,
        -4.745451061101081e-05,
        -3.1161011194140255e-05,
        -3.1161011194140255e-05,
        -3.1161011194140255e-05,
        -3.1161011194140255e-05,
        -3.116096311120378e-05,
        -3.116096311120378e-05,
        -3.116096311120378e-05,
        -3.116096311120378e-05,
        -2.3936289911948183e-05,
        -2.3936289911948183e-05,
        -2.3936289911948183e-05,
        -2.3936289911948183e-05,
        -2.393628503037501e-05,
        -2.393628503037501e-05,
        -2.393628503037501e-05,
        -2.393628503037501e-05,
        -1.663198932563437e-05,
        -1.581817020367027e-05,
        -1.581817020367027e-05,
        -1.1087992883756245e-05,
        -1.1087992883756245e-05,
        -7.978763303982728e-06,
        -7.978763303982728e-06,
        -7.978763303982728e-06,
        -7.978763303982728e-06,
        -7.97876167679167e-06,
        -7.97876167679167e-06,
        -7.97876167679167e-06,
        -7.97876167679167e-06,
        -2.771998220939061e-06,
        -2.771998220939061e-06,
        -1.4138961572840636e-06,
        -1.4138961572840636e-06,
        -1.4138961572840636e-06,
        -1.4138961572840636e-06,
        -1.4138721238216464e-06,
        -1.4138721238216464e-06,
        -1.4138721238216464e-06,
        -1.4138721238216464e-06,
        -4.7129871909468786e-07,
        -4.7129871909468786e-07,
        -4.7129871909468786e-07,
        -4.7129871909468786e-07,
        -4.712907079405488e-07,
        -4.712907079405488e-07,
        -4.712907079405488e-07,
        -4.712907079405488e-07,
        4.811507148005317e-07,
        4.811507148005317e-07,
        4.811507148005317e-07,
        4.811507148005317e-07,
        9.623014296010633e-07,
        9.623014296010633e-07,
        9.623014296010633e-07,
        9.623014296010633e-07,
        1.9246028592021267e-06,
        4.012677014466219e-06,
        4.012677014466219e-06,
        4.012677014466219e-06,
        4.012677014466219e-06,
        4.012743974048166e-06,
        4.012743974048166e-06,
        4.012743974048166e-06,
        4.012743974048166e-06,
        6.450183658722912e-06,
        6.450183658722912e-06,
        6.450183658722912e-06,
        6.450183658722912e-06,
        6.450184566334329e-06,
        6.450184566334329e-06,
        6.450184566334329e-06,
        6.450184566334329e-06,
        8.025354028932438e-06,
        8.025354028932438e-06,
        8.025487948096332e-06,
        8.025487948096332e-06,
        1.9350550976168735e-05,
        1.9350550976168735e-05,
        1.9350550976168735e-05,
        1.9350550976168735e-05,
        1.9350553699002985e-05,
        1.9350553699002985e-05,
        1.9350553699002985e-05,
        1.9350553699002985e-05,
        4.1779316146677584e-05,
        4.1779316146677584e-05,
        4.461690659578001e-05,
        4.461690659578001e-05,
        4.461690659578001e-05,
        4.461690659578001e-05,
        4.461692016190839e-05,
        4.461692016190839e-05,
        4.461692016190839e-05,
        4.461692016190839e-05,
        4.611778769584726e-05,
        4.611778769584726e-05,
        4.611778866220354e-05,
        4.611778866220354e-05,
        8.923381319156002e-05,
        8.923381319156002e-05,
        8.923384032381677e-05,
        8.923384032381677e-05,
        0.00016248534793018287,
        0.00016248534793018287,
        0.00016248534793018287,
        0.00016248534793018287,
        0.0001624853534752421,
        0.0001624853534752421,
        0.0001624853534752421,
        0.0001624853534752421,
        0.00018447115078338905,
        0.00018447115078338905,
        0.00018447115464881417,
        0.00018447115464881417,
        (0.00018836171968557433 + 0j),
        (0.00018836171968557433 + 0j),
        (0.0001883617806066767 + 0j),
        (0.0001883617806066767 + 0j),
        0.0002767067261750836,
        0.0002767067319732213,
        0.00032497069586036575,
        0.00032497069586036575,
        0.0003249707069504842,
        0.0003249707069504842,
        (0.0056609811549201625 + 0j),
        (0.012701001932649118 + 0j),
        (0.012701001935554193 + 0j),
        (0.015368993604463204 + 0j),
    ]
)

# Pre-sorted, operators with indices in ascending order
reference_taylor_bosonic_ops = [
    {(0, 1): "+", (1, 1): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "+"},
    {(0, 1): "-"},
    {(0, 2): "+"},
    {(0, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "+"},
    {(0, 0): "-", (1, 0): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+"},
    {(0, 2): "+", (1, 1): "-"},
    {(0, 1): "+", (1, 2): "-"},
    {(0, 1): "-", (1, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 2): "+"},
    {(0, 2): "+", (1, 0): "-"},
    {(0, 0): "+", (1, 2): "-"},
    {(0, 0): "-", (1, 2): "-"},
    {(0, 0): "+", (1, 1): "+"},
    {(0, 1): "+", (1, 0): "-"},
    {(0, 0): "+", (1, 1): "-"},
    {(0, 0): "-", (1, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "-", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 0): "+"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+"},
    {(0, 0): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 0): "-", (1, 1): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 0): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 2): "+", (1, 2): "+"},
    {(0, 2): "-", (1, 2): "-"},
    {(0, 1): "+", (1, 1): "+"},
    {(0, 1): "-", (1, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 0): "-"},
    {(0, 2): "+", (1, 2): "-"},
    {(0, 1): "+", (1, 1): "-"},
    {},
]

reference_taylor_bosonic_ops_non_loc = [
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+"},
    {(0, 2): "+", (1, 0): "-"},
    {(0, 0): "+", (1, 2): "-"},
    {(0, 0): "-", (1, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 2): "+"},
    {(0, 2): "+", (1, 1): "-"},
    {(0, 1): "+", (1, 2): "-"},
    {(0, 1): "-", (1, 2): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 2): "-"},
    {(0, 2): "+"},
    {(0, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "-", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 0): "+"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 0): "-", (1, 1): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "+"},
    {(0, 1): "+", (1, 0): "-"},
    {(0, 0): "+", (1, 1): "-"},
    {(0, 0): "-", (1, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "+"},
    {(0, 0): "-", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "-", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+"},
    {(0, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 1): "+"},
    {(0, 1): "+", (1, 1): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 0): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 1): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-", (3, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 2): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "+", (3, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 2): "+", (1, 2): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "+"},
    {(0, 1): "+", (1, 0): "-", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "+", (2, 1): "-"},
    {(0, 0): "-", (1, 0): "-", (2, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 0): "-", (2, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "-", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-", (3, 1): "-"},
    {(0, 0): "+", (1, 2): "+", (2, 0): "-", (3, 2): "-"},
    {(0, 0): "+", (1, 1): "+", (2, 0): "-"},
    {(0, 0): "+", (1, 0): "-", (2, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+", (3, 1): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 2): "+", (3, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 2): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "+"},
    {(0, 1): "-", (1, 1): "-", (2, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 1): "-", (3, 2): "-"},
    {(0, 1): "+", (1, 1): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 1): "-", (2, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "+"},
    {(0, 2): "+", (1, 2): "+", (2, 1): "-"},
    {(0, 1): "+", (1, 2): "-", (2, 2): "-"},
    {(0, 1): "-", (1, 2): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "+"},
    {(0, 0): "-", (1, 0): "-"},
    {(0, 1): "+"},
    {(0, 1): "-"},
    {(0, 1): "+", (1, 2): "+", (2, 2): "-"},
    {(0, 2): "+", (1, 1): "-", (2, 2): "-"},
    {(0, 0): "+", (1, 0): "-"},
    {(0, 1): "+", (1, 1): "+"},
    {(0, 1): "-", (1, 1): "-"},
    {(0, 2): "+", (1, 2): "+"},
    {(0, 2): "-", (1, 2): "-"},
    {(0, 1): "+", (1, 1): "-"},
    {(0, 2): "+", (1, 2): "-"},
    {},
]

reference_taylor_bosonic_coeffs_non_loc = [
    2.893562249999999e-13,
    2.893562249999999e-13,
    2.893562249999999e-13,
    2.893562249999999e-13,
    8.680686749999996e-13,
    8.680686749999996e-13,
    8.680686749999996e-13,
    8.680686749999996e-13,
    1.2376149199999996e-12,
    1.2376149199999996e-12,
    1.2376149199999996e-12,
    1.2376149199999996e-12,
    2.8450355499999992e-12,
    2.8450355499999992e-12,
    2.8450355499999992e-12,
    2.8450355499999992e-12,
    3.712844759999999e-12,
    3.712844759999999e-12,
    3.712844759999999e-12,
    3.712844759999999e-12,
    4.580913434999998e-12,
    4.580913434999998e-12,
    4.580913434999998e-12,
    4.580913434999998e-12,
    -4.8314794750903384e-12,
    -4.8314794750903384e-12,
    -4.8314794750903384e-12,
    -4.8314794750903384e-12,
    8.535106649999998e-12,
    8.535106649999998e-12,
    8.535106649999998e-12,
    8.535106649999998e-12,
    -9.662958950180677e-12,
    -9.662958950180677e-12,
    1.7159658574999993e-11,
    1.7159658574999993e-11,
    1.7159658574999993e-11,
    1.7159658574999993e-11,
    2.985885825415369e-11,
    2.985885825415369e-11,
    4.496301739130437e-11,
    4.496301739130437e-11,
    4.496301739130437e-11,
    4.496301739130437e-11,
    5.147897572499998e-11,
    5.147897572499998e-11,
    5.147897572499998e-11,
    5.147897572499998e-11,
    6.001408237499998e-11,
    6.001408237499998e-11,
    6.001408237499998e-11,
    6.001408237499998e-11,
    8.957657476246109e-11,
    8.957657476246109e-11,
    8.992603478260874e-11,
    8.992603478260874e-11,
    1.2970811267867513e-10,
    1.2970811267867513e-10,
    2.545553899999999e-06,
    2.545553899999999e-06,
    2.545553899999999e-06,
    2.545553899999999e-06,
    -2.546772749999999e-06,
    -2.546772749999999e-06,
    4.58588700642727e-06,
    4.58588700642727e-06,
    4.58588700642727e-06,
    4.58588700642727e-06,
    -6.577205824999997e-06,
    -6.577205824999997e-06,
    -6.577205824999997e-06,
    -6.577205824999997e-06,
    7.636661699999997e-06,
    7.636661699999997e-06,
    7.636661699999997e-06,
    7.636661699999997e-06,
    9.17177401285454e-06,
    9.17177401285454e-06,
    -1.0187090999999996e-05,
    -1.0187090999999996e-05,
    -1.2094955774999995e-05,
    -1.2094955774999995e-05,
    -1.2094955774999995e-05,
    -1.2094955774999995e-05,
    -1.5280636499999994e-05,
    -1.5818176791107815e-05,
    -1.5818176791107815e-05,
    -1.9731617474999992e-05,
    -1.9731617474999992e-05,
    -1.9731617474999992e-05,
    -1.9731617474999992e-05,
    2.1436157299999994e-05,
    2.1436157299999994e-05,
    2.1539540474999992e-05,
    2.1539540474999992e-05,
    2.6678452005996513e-05,
    2.6678452005996513e-05,
    -4.0373824499999986e-05,
    -4.0373824499999986e-05,
    -4.0373824499999986e-05,
    -4.0373824499999986e-05,
    -4.745453037332345e-05,
    -4.745453037332345e-05,
    -4.935484374999997e-05,
    -4.935484374999997e-05,
    -4.935484374999997e-05,
    -4.935484374999997e-05,
    6.954709537289269e-05,
    6.954709537289269e-05,
    6.954709537289269e-05,
    6.954709537289269e-05,
    -8.074764899999997e-05,
    -8.074764899999997e-05,
    -8.074764899999997e-05,
    -8.074764899999997e-05,
    8.574462919999997e-05,
    8.574462919999997e-05,
    8.615816189999997e-05,
    8.615816189999997e-05,
    -9.870968749999994e-05,
    -9.870968749999994e-05,
    -9.870968749999994e-05,
    -9.870968749999994e-05,
    (-0.00010500930475000007 + 0j),
    (-0.00010500930475000007 + 0j),
    0.00012861694379999995,
    0.00012923724284999994,
    -0.00013735712212661747,
    -0.00013735712212661747,
    -0.00013735712212661747,
    -0.00013735712212661747,
    0.00013909419074578538,
    0.00013909419074578538,
    0.00014797426349999994,
    0.00014797426349999994,
    0.00014797426349999994,
    0.00014797426349999994,
    -0.00016149529799999995,
    -0.00019741937499999989,
    (0.00022785666259999985 + 0j),
    (0.00022785666259999985 + 0j),
    (0.00023621738279999997 + 0j),
    (0.00023621738279999997 + 0j),
    -0.00027471424425323495,
    -0.00027471424425323495,
    0.0002959485269999999,
    0.0002959485269999999,
    0.0002959485269999999,
    0.0002959485269999999,
    0.00032601776865039307,
    0.00032601776865039307,
    0.0005918970539999998,
    0.0009780533059511793,
    0.0009780533059511793,
    0.0011899339567753636,
    0.0011899339567753636,
    0.0011899339567753636,
    0.0011899339567753636,
    0.002030630140599925,
    0.002030630140599925,
    0.002379867913550727,
    0.002379867913550727,
    (0.005686871390499998 + 0j),
    (0.012734744765599996 + 0j),
    (0.012841643325199995 + 0j),
    (0.015452097370324994 + 0j),
]
expected_coeffs_x_arr = [
    np.array(
        [
            [1.45101915e-16, -8.57929776e-17, -1.96508555e-17, 9.05537757e-18],
            [8.07038739e-17, -2.55246776e-16, 1.26748652e-18, 2.86900384e-17],
            [3.17612270e-16, 9.88792957e-17, -5.06193374e-17, -9.06000149e-18],
        ]
    ),
    np.array(
        [
            [
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    6.28056138e-17,
                    1.10314274e-17,
                    3.85502051e-18,
                    -6.97795726e-18,
                    -4.01282631e-18,
                    3.49879417e-18,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    3.22671619e-17,
                    -4.28062849e-18,
                    2.03758062e-17,
                    -6.73005356e-19,
                    4.70692928e-18,
                    -6.76702152e-18,
                ],
                [
                    -2.82117907e-17,
                    -2.33310517e-17,
                    6.11832626e-18,
                    5.80471685e-18,
                    2.92245962e-18,
                    1.79105471e-18,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
        ]
    ),
    np.array(
        [
            [
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
            ],
            [
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
            ],
            [
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [7.57698351e-18, -3.03499091e-18, -1.59158897e-18, -1.15925948e-18],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
                [
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                    [0.00000000e00, 0.00000000e00, 0.00000000e00, 0.00000000e00],
                ],
            ],
        ]
    ),
]

expected_coeffs_y_arr = [
    np.array(
        [
            [-5.71594742e-12, 1.57896569e-12, 7.77547371e-14, -1.64732892e-14],
            [2.54115137e-02, 7.29476925e-04, -3.66530601e-04, 1.09408770e-05],
            [-2.54115164e-02, -7.29476927e-04, 3.66530612e-04, -1.09408790e-05],
        ]
    ),
    np.array(
        [
            [
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    -7.07234845e-03,
                    4.04421862e-04,
                    -2.39537033e-04,
                    5.09761340e-05,
                    1.42021248e-05,
                    5.21565396e-05,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    7.07234920e-03,
                    -4.04421869e-04,
                    2.39537053e-04,
                    -5.09761379e-05,
                    -1.42021254e-05,
                    -5.21565454e-05,
                ],
                [
                    -2.32180210e-11,
                    -1.11992540e-04,
                    1.11992632e-04,
                    2.04538186e-05,
                    6.00175989e-12,
                    -2.04538352e-05,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
        ]
    ),
]

expected_coeffs_z_arr = [
    np.array(
        [
            [-3.44349847e-02, 2.92644047e-03, 1.04384684e-03, -1.13295425e-04],
            [2.56006019e-02, 5.19670502e-04, -2.97428252e-04, 6.25382340e-06],
            [2.56005992e-02, 5.19670426e-04, -2.97428239e-04, 6.25382238e-06],
        ]
    ),
    np.array(
        [
            [
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    5.72749723e-03,
                    2.41938005e-04,
                    -7.70656829e-05,
                    -2.45454012e-05,
                    -4.27840654e-05,
                    -2.71261299e-04,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
            [
                [
                    5.72749659e-03,
                    2.41937964e-04,
                    -7.70656598e-05,
                    -2.45453921e-05,
                    -4.27840688e-05,
                    -2.71261267e-04,
                ],
                [
                    7.13133100e-04,
                    -1.01350633e-04,
                    -1.01350575e-04,
                    7.71647159e-06,
                    -9.06113606e-06,
                    7.71646583e-06,
                ],
                [
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                    0.00000000e00,
                ],
            ],
        ]
    ),
]
